from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from time import sleep

# 指定chromedriver路径（137版本）
driver_path = "D:\\pycharm\\PM_test\\chromedriver.exe"

# 创建WebDriver实例
service = Service(driver_path)
options = Options()
browser = webdriver.Chrome(service=service, options=options)
browser.maximize_window()
wait = WebDriverWait(browser, 10)

# 登录函数
def login():
    browser.get("http://146.56.217.9:9131")
    browser.find_element(By.ID, "username").send_keys("admin_9131")
    browser.find_element(By.ID, "password").send_keys("admin")
    browser.find_element(By.CLASS_NAME, "button").click()

# 封装工作站搜索函数
def workstation_search(search_data):
    # 跳转至工作站列表界面
    browser.find_element(By.ID, "mainmenua_mrp").click()
    browser.find_element(By.CSS_SELECTOR, ".menu_contenu_workstation_workstation_list .vsmenu").click()
    # 编号
    browser.find_element(By.NAME, "search_ref").send_keys(search_data["ref"])
    # 标签
    browser.find_element(By.NAME, "search_label").send_keys(search_data["label"])
    # 类型（2人员，3机器）
    browser.find_element(By.CSS_SELECTOR, f"select#search_type>option:nth-child({search_data['type']})").click()
    # 状态(2禁用，3生效)
    browser.find_element(By.CSS_SELECTOR, f"select#search_status>option:nth-child({search_data['status']})").click()

    # 搜索按钮
    browser.find_element(By.NAME, "button_search_x").click()

if __name__ == '__main__':
    try:
        login()
        sleep(1)  # 等待页面加载

        # 工作站搜索功能测试数据（3正向+3反向，共6个用例）
        # 测试结果和预期完全一致
        test_data = [
            # --------------------- 正向测试用例 ---------------------
            {
                "case_desc": "精确搜索特定工作站",
                "ref": "WS-20250614-001",  # 假设存在的工作站编号
                "label": "装配工作站-A",
                "type": 2,  # 人员类型
                "status": 3,  # 生效状态
                "expected_result": "搜索结果显示1条记录，与输入条件完全匹配"
            },
            {
                "case_desc": "按类型搜索",
                "ref": "",  # 不填写编号
                "label": "",  # 不填写标签
                "type": 3,  # 机器类型
                "status": 3,  # 生效状态
                "expected_result": "搜索结果显示所有机器类型且生效的工作站"
            },
            {
                "case_desc": "多条件组合搜索",
                "ref": "WS-2025",  # 部分编号
                "label": "测试",  # 部分标签
                "type": 2,  # 人员类型
                "status": 2,  # 禁用状态
                "expected_result": "搜索结果显示编号包含'WS-2025'、标签包含'测试'的禁用人员工作站"
            },

            # --------------------- 反向测试用例 ---------------------
            {
                "case_desc": "无效类型搜索",
                "ref": "",
                "label": "",
                "type": 1,  # 不存在的类型（根据注释，仅允许2和3）
                "status": 3,
                "expected_result": "搜索结果显示0条记录，提示'未找到匹配数据'"
            },
            {
                "case_desc": "无效状态搜索",
                "ref": "",
                "label": "",
                "type": 2,
                "status": 1,  # 不存在的状态（根据注释，仅允许2和3）
                "expected_result": "搜索结果显示0条记录，提示'未找到匹配数据'"
            },
            {
                "case_desc": "不存在的组合条件",
                "ref": "NONEXISTENT",
                "label": "INVALID",
                "type": 3,
                "status": 2,
                "expected_result": "搜索结果显示0条记录，提示'未找到匹配数据'"
            }
        ]
        for data in test_data:
            workstation_search(data)
            sleep(1)  # 等待搜索结果加载
    except Exception as error:
        print("发生错误：", error)
    finally:
        sleep(1)
        browser.quit()